package com.mamingchao.basic.swardToOffer.two;

/**
 * 在一个 n * m 的二维数组中，每一行都按照从左到右递增的顺序排序，每一列都按照从上到下递增的顺序排序。请完成一个高效的函数，输入这样的一个二维数组和一个整数，判断数组中是否含有该整数。
 * 示例:
 *
 * 现有矩阵 matrix 如下：
 *
 * [
 *   [1,   4,  7, 11, 15],
 *   [2,   5,  8, 12, 19],
 *   [3,   6,  9, 16, 22],
 *   [10, 13, 14, 17, 24],
 *   [18, 21, 23, 26, 30]
 * ]
 * 给定 target = 5，返回 true。
 *
 * 给定 target = 20，返回 false
 *
 * 参考同目录的FindNumberIn2DArray里面的实现；主要区别是，FindNumberIn2DArray 是从右上角开始搜索的；每次都可以排除一列或者一行
 * 我自己写的这个例子，是个失败的例子
 */
public class SearchNumInMatrix {

    public static void main(String[] args) {
        final  int rowNum = 3;
        final  int columnNum = 3;
        int[][] matrix = new int[rowNum][columnNum];

        int num = 1;
        for (int i = 0; i < rowNum; i++) {
            for (int j = 0; j < columnNum; j++) {
                matrix[i][j] = num ++;
            }
        }

        boolean result = contain( matrix, 5);
        System.out.println(result);
    }


    private static boolean contain(int[][] matrix, int num) {
        // 行数
        int rowNum = matrix.length;
        // 列数
        int columnNum = matrix[0].length;


        int rowStartIndex = 0;
        int rowEndIndex = rowNum - 1;
        int rowIndex = (rowNum-1)/2;
        while(rowIndex != -1) {
            if (matrix[rowIndex][0] ==  num ) {
                return true;
            }else if (matrix[rowIndex][0] < num ) {
                rowStartIndex = rowIndex;
                rowIndex = getMidIndex(rowStartIndex , rowEndIndex);
            } else {
                rowEndIndex = rowIndex;
                rowIndex = getMidIndex(rowStartIndex, rowEndIndex);
            }
        }




        int columnStartIndex = 0;
        int columnEndIndex = columnNum -1;
        int columnIndex = (columnNum-1)/2;
        while(columnIndex != -1) {
            if (matrix[columnIndex][0] ==  num ) {
                return true;
            }else if (matrix[columnIndex][0] < num ) {
                columnStartIndex = columnIndex;
                columnIndex = getMidIndex(columnStartIndex , columnEndIndex);
            } else {
                columnEndIndex = columnIndex;
                columnIndex = getMidIndex(columnStartIndex, columnEndIndex);
            }
        }



        return false;
    }

    /**
     *
     * @param startIndex
     * @param endIndex
     * @return 返回向下取整的中间index
     */
    private static int getMidIndex(int startIndex, int endIndex) {
        // 循环终止条件
        if (endIndex == startIndex ) {
            return -1;
        } else if(endIndex-startIndex ==1){
            // startIndex 应该在上一轮check过了，所以返回endIndex,换成 endIndex 做check
            return endIndex;
        }

        return (endIndex - startIndex)/2 + startIndex;
    }
}
